from db import student_model

if __name__ == '__main__':
    sm = student_model()
    print('欢迎进入学生信息管理系统')
    while True:
        print('请输入指令，1：查询所有学生信息。 2：添加一个学生 3：删除一个学生 4:修改学生字段 5：退出系统')
        i = input()
        if i == '1':
            sm.query()
        elif i == '2':
            print('请输入学生id')
            stu_id = input()
            print('请输入学生姓名')
            name = input()
            print('请输入学生性别 0 或1')
            sex = input()
            if sex not in ["0", "1"]:
                print('性别输入错误！')
                continue
            print('请输入学生专业')
            major = input()
            sm.insert_student(stu_id, name, int(sex), major)
            print('写入学生信息成功！')
        elif i == '3':
            print('请输入一个学生id')
            stu_id = input()
            sm.del_student(stu_id)
            print('删除成功！')
        elif i == '4':
            print('请输入一个学生id')
            stu_id = input()
            print('请输入字段名')
            colname = input()
            print('请输入值')
            val = input()
            sm.update(stu_id, colname, val)
        elif i == '5':
            print('退出系统!')
            break
        else:
            print('请正确输入指令')
